# -*- coding: utf-8 -*-
"""
---------------------------------------
@File    ：add_classify.py
@Author  ：liyunhui
@Date    ：2024/9/16 下午4:51 
@explain : 文件说明
---------------------------------------
"""
from config import config
from methods.redis_connect import get_redis_cli

redis_cli = get_redis_cli()

classify_data = {
    "youtube-stream": {
        "ip_type": "国外",  # 该分类代理允许的代理分类
        "max_use_per_min": 3,  # 该分类爬虫每分钟最大使用次数
        "check_method": "",  # 该分类爬虫检测请求方法
        "check_url": "",  # 该分类爬虫IP检测URL
        "check_form_data": "",  # 该分类爬虫检测请求参数，可以为空
        "ip_check_time": 10 * 60,  # 该分类过期检测时间区间，该IP的更新时间与当前时间间隔超过这个时间，默认为过期
    }
}

classify_key = config.proxy_classify_key

for classify_name, classify_info in classify_data.items():
    key = classify_key + f':{classify_name}'
    # 将 classify_info 字典中的键值对存储到 Redis 哈希中
    for field, value in classify_info.items():
        redis_cli.hset(key, field, value)
